package com.qianfeng.smartdevice.websocket;//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼                  BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  
//  


import com.qianfeng.smartdevice.events.DevicePowerCommandEvent;
import com.qianfeng.smartdevice.events.DeviceStatusChangeEvent;
import com.qianfeng.smartdevice.events.HeartBeatCheckEvent;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.event.EventListener;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Created by Jackiechan on 2022/1/10/20:20
 *
 * @author Jackiechan
 * @version 1.0
 * @since 1.0
 */
public class MyBaseHandler extends TextWebSocketHandler {
    //保存所有客户端的map
    private Map<String, WebSocketSession> allClients = new ConcurrentHashMap<>();
    private Map<String, Long> allClientsHeartBeatTime = new ConcurrentHashMap<>();

    private ApplicationContext context;


    protected Map<String, WebSocketSession> getAllClients() {
        return allClients;
    }

    @Autowired
    public void setContext(ApplicationContext context) {
        this.context = context;
    }

    protected ApplicationContext getContext() {
        return context;
    }

    /**
     * 当连接建立的时候执行
     * @param session
     * @throws Exception
     */
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        String uri = session.getUri().toString();//获取websocket请求的地址
        String id = uri.substring(uri.lastIndexOf("/") + 1);//我们约定,地址的最后一位就是请求的设备的编号
        WebSocketSession currentSession = allClients.get(id);//查看下当前id是不是之前建立㢧链接
        if (currentSession != null&&currentSession.isOpen()) {
            currentSession.close();
        }
        String ip = session.getRemoteAddress().getHostName();//访问的ip
        session.getAttributes().put("ip", ip);
        session.getAttributes().put("id", id);//保存下数据,方便后面使用
        allClients.put(id, session);//保存当前的会话
        context.publishEvent(new DeviceStatusChangeEvent(id, 1L,ip));
        System.err.println("类PowerCommandHandler中的方法afterConnectionEstablished执行了"+id);
        allClientsHeartBeatTime.put(id, System.currentTimeMillis());//保存连接的时间
    }

    /**
     * 服务器收到客户端发来的数据的时候执行
     * @param session
     * @param message
     * @throws Exception
     */
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        allClientsHeartBeatTime.put((String) session.getAttributes().get("id"), System.currentTimeMillis());//更新连接的时间
        System.err.println("收到设备"+session.getAttributes().get("id")+"发来的数据" + message.getPayload());
    }

    /**
     * 数据传输错误的时候执行
     * @param session
     * @param exception
     * @throws Exception
     */
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        super.handleTransportError(session, exception);
    }

    /**
     * 连接关闭的时候执行
     * @param session
     * @param status
     * @throws Exception
     */
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        //更新状态为1
        String id = (String) session.getAttributes().get("id");
        allClients.remove(id);//从客户端集合中移除当前数据
        allClientsHeartBeatTime.remove(id);//删除心跳时间数据
        context.publishEvent(new DeviceStatusChangeEvent(id, 2L, null));
    }



    @EventListener
    @Async
    public void onEvent(HeartBeatCheckEvent event) {
        System.err.println("检查设备在线的事件执行了");
        //收到检查心跳的事件了,遍历所有的设备,然后看看最后的时间超出返回就关闭会话
        ConcurrentHashMap<String, Long> concurrentHashMap = new ConcurrentHashMap<>();
        concurrentHashMap.putAll(allClientsHeartBeatTime);
        CompletableFuture.runAsync(()->{
            concurrentHashMap.forEach((id,time)->{
                //获取到了每一个设备的id和最后一个时间
                if (System.currentTimeMillis()-time>=60000) {
                    //超时了
                    try {
                        System.err.println("设备:" + id + "超时了,断开会话");
                        //allClients.get(id).close();//关闭
                        WebSocketSession session = allClients.get(id);
                        if (session != null&&session.isOpen()) {
                            session.close();
                        }
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            });
        });
    }
}
